#ifndef KPEXTOPIMGF32C1AVERAGE_H
#define KPEXTOPIMGF32C1AVERAGE_H

#include "kpextVkContext.h"
#include "kpextBaseCompute.h"

namespace kpext {

/**
 * @brief 单通道32位单精度浮点数图像权重累加，实现平均操作
 */
class OpImgF32C1Average : public BaseCompute{
public:
    explicit OpImgF32C1Average(kpextVkContext * ctx);

    virtual ~OpImgF32C1Average() override;

    /**
     * @brief  重新构建计算管线及相关资源
     * @param cols  输入二维图像的列数
     * @param rows  输入二维图像的行数
     * @param weight 每次执行的权重，一般N帧平均设置为 1.0f/N
     * @param inbuf  输入缓冲
     * @param outbuf  输出缓冲
     */
    virtual void rebuild(int cols,
                         int rows,
                         float weight,
                         std::shared_ptr<BaseBuffer> inbuf,
                         std::shared_ptr<BaseBuffer> outbuf
                         );

protected:

    /**
     * @brief 计算着色器的work group size
     */
    std::array<uint32_t, 3> m_workGroupSize;

private:
    using BaseCompute::rebuild;

};  // class kpext::OpImgF32C1Average

} // namespace kpext

#endif // KPEXTOPIMGF32C1AVERAGE_H
